package com.ragdollcat.thirdaicodegenerate.model.entity;

import com.ragdollcat.thirdaicodegenerate.enums.FileType;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.ArrayList;
import java.util.List;

/**
 * 目录/文件节点实体类
 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public class FileNode {
    // 节点名称（文件名或目录名）
    private String name;
    // 节点类型（目录/文件）
    private FileType type;
    // 子节点列表（仅目录有，文件的子节点为空）
    private List<FileNode> children;

    // 构造方法（文件节点：无children）
    public FileNode(String name, FileType type) {
        this.name = name;
        this.type = type;
        this.children = new ArrayList<>();  // 初始化空列表，避免NPE
    }

    /**
     * 为目录添加子节点（仅目录可调用，文件调用会抛异常）
     *
     * @param child 子节点
     */
    public void addChild(FileNode child) {
        if (this.type != FileType.DIRECTORY) {
            throw new UnsupportedOperationException("文件节点不能添加子节点");
        }
        this.children.add(child);
    }
}